# -*- coding: utf-8 -*-
"""
Created on Fri Jun 14 11:34:33 2024

@author: xiaok
"""
import torch

from torchsummary import summary

from EEGNet import EEGNet

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

ch_num = 22
time_points = 257

input_size = (1, ch_num, time_points) #1716 Trainable Parameters (As mentioned in paper)
# input_size = (1, 22, 257) #1716 Trainable Parameters (As mentioned in paper)
eegnet_model = EEGNet(chans=ch_num, time_points=time_points).to(device)
summary(eegnet_model, input_size)





